package com.catering.controller;

import com.catering.pojo.DTable;
import com.catering.service.ITableService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 桌位管理控制器
 */
@RestController
@RequestMapping("/table")
public class TableController {

    @Autowired
    private ITableService tableService;

    /**
     * 根据区域代码获取桌位
     * @param areaCode 区域代码
     * @return 桌位列表
     */
    @GetMapping("/area/{areaCode}")
    public ResponseEntity<List<DTable>> getTablesByArea(@PathVariable String areaCode) {
        List<DTable> tables = tableService.findTablesByArea(areaCode);
        return ResponseEntity.ok(tables);
    }

    /**
     * 获取所有区域及其桌位
     * @return 区域与桌位映射
     */
    @GetMapping("/all-areas")
    public ResponseEntity<Map<String, List<DTable>>> getAllAreasWithTables() {
        Map<String, List<DTable>> result = tableService.findAllAreasWithTables();
        return ResponseEntity.ok(result);
    }

    /**
     * 获取所有区域代码
     * @return 区域代码列表
     */
    @GetMapping("/areas")
    public ResponseEntity<List<String>> getAllAreas() {
        List<String> areas = tableService.findAllAreas();
        return ResponseEntity.ok(areas);
    }

    /**
     * 添加桌位
     * @param table 桌位信息
     * @return 添加结果
     */
    @PostMapping
    public ResponseEntity<?> addTable(@RequestBody DTable table) {
        // 验证区域代码
        if (table.getAreaCode() == null || !table.getAreaCode().matches("[A-F]")) {
            return ResponseEntity.badRequest().body("无效的区域代码");
        }

        // 验证桌位编号格式
        String tableNumber = table.getTableNumber();
        if (tableNumber == null || !tableNumber.matches(table.getAreaCode() + "\\d{2}")) {
            return ResponseEntity.badRequest().body("桌位编号格式不正确");
        }

        // 验证桌位名称格式
        String tableName = table.getTableName();
        if (tableName == null || !tableName.startsWith(table.getAreaCode() + "区-")) {
            return ResponseEntity.badRequest().body("桌位名称格式不正确");
        }

        // 验证桌位编号是否已存在
        DTable existingTable = tableService.findByTableNumber(tableNumber);
        if (existingTable != null) {
            return ResponseEntity.badRequest().body("桌位编号已存在");
        }

        // 设置默认状态
        if (table.getStatus() == null) {
            table.setStatus(0); // 默认为空闲状态
        }

        boolean success = tableService.addTable(table);
        if (success) {
            return ResponseEntity.ok().body("桌位添加成功");
        } else {
            return ResponseEntity.badRequest().body("桌位添加失败");
        }
    }

    /**
     * 更新桌位状态
     * @param tableNumber 桌位编号
     * @param status 新状态
     * @return 更新结果
     */
    @PutMapping("/{tableNumber}/status")
    public ResponseEntity<?> updateTableStatus(@PathVariable String tableNumber, @RequestParam Integer status) {
        boolean success = tableService.updateTableStatus(tableNumber, status);
        if (success) {
            return ResponseEntity.ok().body("桌位状态更新成功");
        } else {
            return ResponseEntity.badRequest().body("桌位不存在或状态更新失败");
        }
    }

    /**
     * 删除桌位
     * @param tableNumber 桌位编号
     * @return 删除结果
     */
    @DeleteMapping("/{tableNumber}")
    public ResponseEntity<?> deleteTable(@PathVariable String tableNumber) {
        boolean success = tableService.deleteTable(tableNumber);
        if (success) {
            return ResponseEntity.ok().body("桌位删除成功");
        } else {
            return ResponseEntity.badRequest().body("桌位不存在或删除失败");
        }
    }

    /**
     * 获取桌位统计信息
     * @return 统计信息
     */
    @GetMapping("/statistics")
    public ResponseEntity<Map<String, Object>> getTableStatistics() {
        Map<String, Object> statistics = new HashMap<>();

        // 获取所有区域
        List<String> areas = tableService.findAllAreas();
        Map<String, Integer> areaTableCounts = new HashMap<>();
        int totalTables = 0;
        int occupiedTables = 0;

        // 统计每个区域的桌位数量
        for (String area : areas) {
            List<DTable> tables = tableService.findTablesByArea(area);
            int count = tables.size();
            areaTableCounts.put(area, count);
            totalTables += count;

            // 统计已占用桌位
            for (DTable table : tables) {
                if (table.getStatus() == 1) {
                    occupiedTables++;
                }
            }
        }
        statistics.put("totalTables", totalTables);
        statistics.put("occupiedTables", occupiedTables);
        statistics.put("availableTables", totalTables - occupiedTables);
        statistics.put("areaTableCounts", areaTableCounts);
        return ResponseEntity.ok(statistics);
    }


}
